tensor([[[[[0.0000, 0.0000],
           [0.5600, 0.0000]],

          [[0.0000, 0.2400],
           [0.2400, 0.5500]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0900, 0.0700],
           [0.0900, 0.1100]]]],



        [[[[0.4900, 0.0000],
           [0.0000, 0.5600]],

          [[0.5400, 0.2800],
           [0.2800, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.5000]],

          [[0.1500, 0.1200],
           [0.1800, 0.1900]]]],



        [[[[0.0000, 0.4500],
           [0.4400, 0.4400]],

          [[0.4600, 0.2200],
           [0.2200, 0.4500]]],


         [[[1.0000, 0.4800],
           [0.4800, 0.0000]],

          [[0.3600, 0.3400],
           [0.3800, 0.3700]]]],



        [[[[0.5100, 0.5500],
           [0.0000, 0.0000]],

          [[0.0000, 0.2600],
           [0.2600, 0.0000]]],


         [[[0.0000, 0.5200],
           [0.5200, 0.5000]],

          [[0.4000, 0.4700],
           [0.3500, 0.3300]]]]], grad_fn=<RoundBackward1>)